package demo2;

import java.util.Arrays;

public class TestHeap {
    public int[] elem;
    public int usedSize;


    public TestHeap() {
        this.elem = new int[10];
    }

    public void initElem(int[] array) {
        for (int i = 0; i < array.length; i++) {
            this.elem[i] = array[i];
            this.usedSize++;
        }
    }

    //创建堆
    public void createHeap() {
        for(int parent=(this.usedSize-1-1)/2;parent>=0;parent--){
            siftDown(parent,this.usedSize);//从上往下创建堆
        }
    }
    /**
     * @param parent 每棵子树调整的时候 的 起始位置
     * @param usedSize 判断 每棵子树什么时候 调整 结束
     */

    private void siftDown(int parent, int usedSize) {
        //创建大根堆
        int child=2*parent+1;
        while(child<usedSize){
            //1.找左右子树的最大值
            if(child+1<usedSize&&elem[child]<elem[child+1]){
                child++;
            }
            if(elem[parent]<elem[child]){
                swap(elem,parent,child);
                parent=child;
                child=2*parent+1;
            }else{
                break;
            }
        }
    }

    private void swap(int[] elem,int i,int j) {
        int tmp=j;
        i=j;
        j=tmp;

    }

    public void offer(int val) {
        //1.检查是否满
        if(isFull()){
            elem=Arrays.copyOf(this.elem,2*elem.length);
        }
        elem[usedSize]=val;
        siftUp(usedSize);
        usedSize++;

    }

    private void siftUp(int child) {
        //从下往上，创建大根堆
        int parent=(child-1)/2;
        while(parent>=0){
            if(elem[child]>elem[parent]){
                swap(elem,child,parent);
                child=parent;
                parent=(child-1)/2;
            }else{
                break;
            }
        }
    }

    public boolean isFull() {
        return usedSize == elem.length;
    }

    public int poll() {
        if(isEmpty()){
            return -1;
        }
        int val=elem[0];
        swap(elem,0,usedSize-1);
        //因为要拿出一个元素，故有效值就变为了usedsize-1,拿出的那个元素不用管
        siftDown(0,usedSize-1);
        usedSize--;
        return val;
    }
    public boolean isEmpty() {
        return usedSize == 0;
    }

    public int peek() {
        if (isEmpty()){
            return -1;
        }
        return elem[0];

    }

    //O(N) + o(n*logN) ->  o(n*logN)
    public void heapSort() {
        //在堆本身排序,从小打到排
        int end=this.usedSize-1;
        while(end>0){
            swap(elem,0,end);
            siftDown(0,end);
            end--;

        }

    }

}
